#include<iostream>
#include<memory>//智能指针
#include"Route.hpp"//路由器
#include "UdpServer.hpp" // 网络通信的功能
#include "ThreadPool.hpp"

using namespace ThreadPoolModule;

using task_t = std::function<void()>;

// ./udpserver port
int main(int argc, char* argv[])
{
    if(argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port" << std::endl;
        return 1;
    }
    // std::string ip = argv[1];
    uint16_t port = std::stoi(argv[1]);
    Enable_Console_Log_Strategy();

    //1. 路由服务
    Route r;

    //2. 线程池
    //<task_t>的作用是显式指定ThreadPool实例化时的模板参数，表明这个线程池处理的任务类型是task_t
    auto tp = ThreadPool<task_t>::GetInstance();

    //3. 网络服务器对象，提供通信功能
    //UdpServer(uint16_t port, func_t func)                    //r - 路由，tp - 线程池
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp]
        (int sockfd, const std::string &message, InetAddr &peer){
        task_t t = std::bind(&Route::MessageRoute, &r, sockfd, message, peer);
        //bind本质是返回一个仿函数对象 -- 看博客
        tp->Enqueue(t);
    });

    usvr->Init();
    usvr->Start();

    return 0;
}